package com.briup.oagg.model.service;

import com.briup.oagg.model.bean.basic.RegressionTrainResult;
import com.baomidou.mybatisplus.extension.service.IService;
import com.briup.oagg.model.bean.op.TaskResult;
import com.briup.oagg.model.web.vm.RegressionTrainResultVM;

import java.util.List;

/**
 * 回归训练评估结果信息业务逻辑接口
 *
 * @author briup
 */
public interface IRegressionTrainResultService extends IService<RegressionTrainResult> {
    /**
     * 根据训练编号查询对应的训练结果
     * @param trainId 训练编号
     * @return 训练结果
     */
    RegressionTrainResultVM findByTrainId(String trainId);

    /**
     * 新增训练结果
     * @param result  训练结果
     */
    void addTrainResult(TaskResult result);

    /**
     * 根据模型编号查询所有的训练信息
     * @param modelId 模型编号
     * @return 训练结果列表
     */
    List<RegressionTrainResultVM> findByModelId(String modelId);
}
